sum in python while loop